from pydantic import BaseModel, Field


class DownloadResponse(BaseModel):
    success: bool
    saved_path: str
    size: float = Field(default=0, description="Total size of the files in testing data")
    folder_number: int = Field(..., description="The folder number in the testing data")

class DownloadBody(BaseModel):
    host_name: str = Field(..., description="Robot ip address")